Task Context Shapes Component Roles in Language Models
任务上下文塑造语言模型中组件角色
🧭 语言模型组件角色的任务上下文依赖性
University of Pennsylvania · Yale University
任务上下文改变共享注意力头响应的事实及其对后续计算的影响,线性响应图可预测干预效果
📌 工作参考价值(4 点)
- 在多任务电商/搜索 LLM 中,共享注意力头的角色随任务 Prompt 改变:可测量不同业务任务(query 改写 vs 推荐理由生成)下各头的影响分布,识别任务关键头,指导稀疏微调或任务特定低秩适配,避免全参更新。
- 线性响应图可作为轻量任务行为代理:为不同业务 Prompt 构建组件间线性映射,用早期激活变化预测后续层响应,降低线上任务切换时的行为漂移监控成本。
- 激活编辑切换行为而不改指令:在可控生成或 Agent 风格切换场景,只编辑早期内部激活即可让模型读取不同事实/知识,无需重写 Prompt,可用于无感干预或 A/B 隔离测试。
- 但整体属于机制可解释性研究,业务直接收益有限;其干预测量与影响分数方法可迁移到诊断推荐/搜索任务间泛化失败原因,但需自行设计干预目标。